package com.atlige.gzmall.auth.controller;

import com.alibaba.fastjson.TypeReference;
import com.atlige.common.constant.AuthServerConstant;
import com.atlige.common.exception.GzmallCodeEnum;
import com.atlige.common.utils.R;
import com.atlige.common.vo.MemberRespVo;
import com.atlige.gzmall.auth.feign.MemberFeignService;
import com.atlige.gzmall.auth.feign.ThirdPartFeignService;
import com.atlige.gzmall.auth.vo.UserLoginVo;
import com.atlige.gzmall.auth.vo.UserRegistVo;
import org.apache.commons.lang3.time.DateUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.util.StringUtils;
import org.springframework.validation.BindingResult;
import org.springframework.validation.FieldError;
import org.springframework.validation.ObjectError;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;

import javax.servlet.http.HttpSession;
import javax.validation.Valid;
import javax.validation.constraints.NotEmpty;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * @author Jeremy Li
 * @data 2021/3/31 - 16:01
 */
@Controller
public class LoginController {
    @Autowired
    ThirdPartFeignService thirdPartFeignService;
    @Autowired
    StringRedisTemplate redisTemplate;
    @Autowired
    MemberFeignService memberFeignService;

    /**
     * 发送验证码
     * @param phone
     * @return
     */
    @ResponseBody
    @GetMapping("/sms/sendcode")
    public R sendCode(@RequestParam("phone") String phone){
        // TODO 1、接口防刷
        String redisCode = redisTemplate.opsForValue().get(AuthServerConstant.SMS_CODE_CACHE_PREFIX + phone);
        if(!StringUtils.isEmpty(redisCode)){
            long l = Long.parseLong(redisCode.split("_")[1]);
            if (System.currentTimeMillis() - l < 60000){
                // 60s内不能重复发送
                return R.error(GzmallCodeEnum.SMS_CODE_EXCEPTION.getCode(), GzmallCodeEnum.SMS_CODE_EXCEPTION.getMessage());
            }
        }
        String code = UUID.randomUUID().toString().substring(0, 5);
        String substring = code + "_" + System.currentTimeMillis();
        // 2、验证码再次校验，redis存key-phone，value-code，防止同一个手机号在60s内再次发送验证码
        // sms:code:132xxxxxxxx -> 789987
        redisTemplate.opsForValue().set(AuthServerConstant.SMS_CODE_CACHE_PREFIX + phone, substring, 10, TimeUnit.MINUTES);
        thirdPartFeignService.sendCode(phone, code);

        return R.ok();
    }

    /**
     * //TODO 重定向携带数据，分布式下的session问题
     * 只要跳到下一个页面取出数据后，session里面的数据就会删掉
     * 利用session原理，将数据放在session中
     * RedirectAttributes redirectAttributes 模拟重定向携带数据
     * @param userRegistVo
     * @param result
     * @param redirectAttributes
     * @return
     */
    @PostMapping("/regist")
    public String regist(@Valid UserRegistVo userRegistVo, BindingResult result, RedirectAttributes redirectAttributes){
        if (result.hasErrors()){
            // 校验出错，转发到注册页
            Map<String, String> errors = result.getFieldErrors().stream().collect(Collectors.toMap(FieldError::getField, FieldError::getDefaultMessage));
            redirectAttributes.addFlashAttribute("errors", errors);
            /*
                Request method 'POST' not support
                用户注册 -> /regist[post请求] -> 转发/reg.html(路径映射默认是get方式)
             */
            return "redirect:http://auth.gzmall.com/reg.html";
        }
        // 真正注册，调用远程服务进行注册
        // 1、校验验证码
        String code = userRegistVo.getCode();
        String redisCode = redisTemplate.opsForValue().get(AuthServerConstant.SMS_CODE_CACHE_PREFIX + userRegistVo.getPhone());
        if (!StringUtils.isEmpty(redisCode)){
            if (code.equalsIgnoreCase(redisCode.split("_")[0])){
                // 删除验证码,令牌机制
                redisTemplate.delete(AuthServerConstant.SMS_CODE_CACHE_PREFIX + userRegistVo.getPhone());
                // 验证码通过，真正注册，调用远程服务进行注册
                R registResult = memberFeignService.regist(userRegistVo);
                if (registResult.getCode() == 0){
                    // 成功
                    return "redirect:http://auth.gzmall.com/login.html";
                }else {
                    // 失败
                    Map<String, String> errors = new HashMap<>();
                    errors.put("msg", registResult.getData("msg", new TypeReference<String>(){}));
                    redirectAttributes.addFlashAttribute("errors", errors);
                    return "redirect:http://auth.gzmall.com/reg.html";
                }
            }else {
                Map<String, String> errors = new HashMap<>();
                errors.put("code", "验证码错误");
                redirectAttributes.addFlashAttribute("errors", errors);
                return "redirect:http://auth.gzmall.com/reg.html";
            }
        }else {
            Map<String, String> errors = new HashMap<>();
            errors.put("code", "验证码错误");
            redirectAttributes.addFlashAttribute("errors", errors);
            return "redirect:http://auth.gzmall.com/reg.html";
        }
        // 重定向到首页或登录页
    }

    @GetMapping("/login.html")
    public String loginPage(HttpSession session){
        Object attribute = session.getAttribute(AuthServerConstant.LOGIN_USER);
        if (attribute == null){
            // 没登陆
            return "login";
        }else {
            return "redirect:http://gzmall.com";
        }
    }

    @PostMapping("/login")
    public String login(UserLoginVo loginVo, RedirectAttributes redirectAttributes,
                        HttpSession session){
        // 调用远程服务登录
        R login = memberFeignService.login(loginVo);
        if (login.getCode() == 0){
            // 成功
            MemberRespVo data = login.getData(new TypeReference<MemberRespVo>() {
            });
            session.setAttribute(AuthServerConstant.LOGIN_USER, data);
            return "redirect:http://gzmall.com";
        }else {
            Map<String, String> errors = new HashMap<>();
            errors.put("msg", login.getData("msg", new TypeReference<String>(){}));
            redirectAttributes.addFlashAttribute("errors", errors);
            return "redirect:http://auth.gzmall.com/login.html";
        }
    }
}
